problem solving using python